#pragma once

#include <vector>
#include <cassert>

#include "Util.hpp"

namespace bin
{
    #define DEFAULT_BUFFER_SIZE (1 * 1024 * 1024)
    #define THRESHOLD_BUFFER_SIZE (8*1024*1024)
    #define INCREMENT_BUFFER_SIZE (1*1024*1024)
    class Buffer
    {
    public:
        Buffer()
        :_buffer(DEFAULT_BUFFER_SIZE)
        ,_writer_idx(0)
        ,_reader_idx(0)
        {}

        // 向缓冲区写入数据
        void push(const char *data, size_t len)
        {
            //缓冲区剩余时间不够：
            //1.扩容
            //2.阻塞/返回false

            //1.固定大小，超出容量直接返回
            // if(len > writeAbleSize())
            //     return;

            //2.动态空间，用于极限性能测试
            ensureEnoughSize(len);

            //1.将数据拷贝进缓冲区
            std::copy(data, data + len, &_buffer[_writer_idx]);
            //2.将当前写入位置向后偏移
            moveWriter(len);
        }

        // 返回可写取数据的长度
        size_t writeAbleSize()
        {
            //对于扩容思路，不存在可写空间大小，因为总是可写
            //本接口仅为固定大小缓冲区提供
            return (_buffer.size() - _writer_idx);
        }

        // 返回可读数据的起始
        const char *begin()
        {
            return (&_buffer[_reader_idx]);
        }

        // 返回可读取数据的长度
        size_t readAbleSize()
        {
            //因为缓冲区不是循环缓冲区，因此只会向后写。
            return (_writer_idx - _reader_idx);
        }

        // 对读指针进行向后偏移操作
        void moveReader(size_t len)
        {
            assert(len <= readAbleSize());
            _reader_idx += len;
        }

        // 重置读写位置，初始化缓冲区
        void reset()
        {
            _reader_idx = 0;
            _writer_idx = 0;
        }

        // 交换缓冲区
        void swap(Buffer &buffer)
        {
            _buffer.swap(buffer._buffer);
            std::swap(_reader_idx, buffer._reader_idx);
            std::swap(_writer_idx, buffer._writer_idx);
        }

        // 判断缓冲区中有没有数据
        bool empty()
        {
            return (_reader_idx == _writer_idx);
        }

    private:
        //空间扩容]
        void ensureEnoughSize(size_t len)
        {
            if(len <= writeAbleSize())
                return;
            
            size_t new_size = 0;
            if(_buffer.size() < THRESHOLD_BUFFER_SIZE)
            {
                new_size = _buffer.size() * 2;
            }
            else
            {
                new_size = _buffer.size() + INCREMENT_BUFFER_SIZE;
            }
            _buffer.resize(new_size);

        }

        // 对写指针进行向后偏移操作
        void moveWriter(size_t len)
        {
            assert((len + _writer_idx) <= _buffer.size());
            _writer_idx += len;
        }

    private:
        std::vector<char> _buffer;
        size_t _reader_idx;
        size_t _writer_idx;
    };
}